from langchain_openai import ChatOpenAI

api_key = "xxxx"
glm = ChatOpenAI(
    temperature=0.01,
    # model="glm-3-turbo",
    model="GLM-4",
    # model="GLM-4-AirX",
    openai_api_key=api_key,
    openai_api_base="https://open.bigmodel.cn/api/paas/v4/",
)


glm_lite = ChatOpenAI(
    temperature=0.01,
    model="GLM-4-Air",
    openai_api_key=api_key,
    openai_api_base="https://open.bigmodel.cn/api/paas/v4/",
)
